import os
import sys
import pandas as pd
import datetime
from MainConsolidator import MainConsolidator

class handler(object):
    @staticmethod
    def module_name():
        return 'ЛПК. Рулон'
    
    @staticmethod
    def handle(row:pd.Series,consolidator:MainConsolidator)->pd.Series:
        product_up=None
        try:
            row["Продукт УП"]=pd.NA
            steel_mark = row["Класс прочности <Производство>"] if pd.isna(row["Марка стали <Производство>"]) else row["Марка стали <Производство>"]
            cl_proch=consolidator.scalar_in_guid("Класс прочности по марке стали",f"`Марка стали`=='{steel_mark}'","Класс прочности") 
            if not pd.isna(steel_mark) and pd.isna(cl_proch): cl_proch=steel_mark
            row["Марка стали <Производство>"]=steel_mark
            row["Класс прочности <Производство>"]=cl_proch
            steel_mark2=consolidator.scalar_in_guid("Маппинг класса прочности",f"`ИсX Класс прочности`=='{steel_mark}'", "Результ Класс прочности")
            cl_proch2=consolidator.scalar_in_guid("Маппинг класса прочности",f"`ИсX Класс прочности`=='{cl_proch}'", "Результ Класс прочности")    
            if pd.notna(steel_mark2): row["Марка стали <Производство>"]=steel_mark2
            if pd.notna(cl_proch2): row["Марка стали <Производство>"]=cl_proch2
            if pd.isna(row["Марка стали <Производство>"]) : row["ГМСПС"]="ГУ"
            else : row["ГМСПС"]=pd.NA
            # # Уточнение ГМСПС
            gmsps=consolidator.scalar_in_guid("Группы марок стали по потребительским свойствам", f"`Марка стали`=='{row['Марка стали <Производство>']}'", "Группа марок")
            if pd.notna(gmsps): row["ГМСПС"]=gmsps
            oto = consolidator.scalar_in_guid("Признак по НТД", 
                                            f"`Производство`=='{row['Производство2']}' and `НТД качества (трубы стальные) <Производство>`=='{row['НТД качества (прокат) <Производство>']}'",  
                                            "Доп параметр1")    
            if pd.notna(oto): row["ГМСПС"]=oto
            # row[['Толщина, мм <Производство>']].fillna("")
            w=row['Толщина, мм <Производство>']
            # print(f"w={w} ({row['Толщина, мм <Производство>']}) : {pd.isna(row['Толщина, мм <Производство>'])}")
            thickness_range = consolidator.scalar_in_guid("Параметры по Продукту Прокат", 
                                                    f"`Производство`=='Рулон' and `Продукт`=='РУЛОН' and `Мин`<={w}<=`Макс` and `Доп параметр1` == '{row['ГМСПС']}'", 
                                                    "Диапазон") 
        
            product_up="РУЛОН "+row["ГМСПС"]
            if pd.notna(row["ГМСПС"]) and "ГУ" in str(row["ГМСПС"]) and pd.notna(thickness_range) : 
                product_up = f"РУЛОН {thickness_range} ГУ"
            if str(row["Признак БЗ"])=="БЗ": 
                product_up = "РУЛОН Р БЗ"
            row["error"] = pd.NA
        except Exception as exp: 
            row["error"] = f"{exp}"
        finally:
            row["Продукт УП"]=product_up
            row["Время укрупнения"]=datetime.datetime.now()
            return row
        
    # @staticmethod
    def debug_handle(self,inputDf:pd.DataFrame,consolidator:MainConsolidator,module:str=None)->pd.DataFrame:
        try:
            if module is None: return inputDf.apply(lambda row: self.handle(row,consolidator), axis=1)
            else: return inputDf[inputDf["Производство2"]==module].apply(lambda row: self.handle(row,consolidator), axis=1)
        except: raise
    
    def get_compiled(self):
        code:str=f"print('{__file__}')"
        try:
            with open(__file__,encoding="utf-8",mode="r") as f:
                code=f.read()
        except Exception as exp: 
            code=f"print('error: {exp}')"
        finally:
            cc = compile(code,"","exec")
            return cc

if __name__=="__roll__":
    consolidator:MainConsolidator
    record:pd.Series
    result:pd.Series
    result = handler.handle(record,consolidator)

